home *** CD-ROM | disk | FTP | other *** search
/ PC World Komputer 2010 April / PCWorld0410.iso / hity wydania / Ubuntu 9.10 PL / karmelkowy-koliberek-desktop-9.10-i386-PL.iso / casper / filesystem.squashfs / usr / share / pyshared / DistUpgrade / utils.py < prev    next >
Encoding:
Python Source  |  2009-11-02  |  9.1 KB  |  292 lines
  1. # utils.py 
  2. #  
  3. #  Copyright (c) 2004-2008 Canonical
  4. #  
  5. #  Author: Michael Vogt <mvo@debian.org>
  6. # 
  7. #  This program is free software; you can redistribute it and/or 
  8. #  modify it under the terms of the GNU General Public License as 
  9. #  published by the Free Software Foundation; either version 2 of the
  10. #  License, or (at your option) any later version.
  11. # 
  12. #  This program is distributed in the hope that it will be useful,
  13. #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15. #  GNU General Public License for more details.
  16. # 
  17. #  You should have received a copy of the GNU General Public License
  18. #  along with this program; if not, write to the Free Software
  19. #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  20. #  USA
  21.  
  22. from gettext import gettext as _
  23. import locale
  24. import os
  25. import os.path
  26. import apt_pkg
  27. import urllib2
  28. import subprocess
  29. from stat import *
  30.  
  31. def lsmod():
  32.   " return list of loaded modules (or [] if lsmod is not found) "
  33.   modules=[]
  34.   # FIXME raise?
  35.   if not os.path.exists("/sbin/lsmod"):
  36.     return []
  37.   p=subprocess.Popen(["/sbin/lsmod"], stdout=subprocess.PIPE)
  38.   lines=p.communicate()[0].split("\n")
  39.   # remove heading line: "Modules Size Used by"
  40.   del lines[0]
  41.   # add lines to list, skip empty lines
  42.   for line in lines:
  43.     if line:
  44.       modules.append(line.split()[0])
  45.   return modules
  46.  
  47.  
  48. def check_and_fix_xbit(path):
  49.   " check if a given binary has the executable bit and if not, add it"
  50.   if not os.path.exists(path):
  51.     return
  52.   mode = S_IMODE(os.stat(path)[ST_MODE])
  53.   if not ((mode & S_IXUSR) == S_IXUSR):
  54.     os.chmod(path, mode | S_IXUSR)
  55.  
  56. def country_mirror():
  57.   " helper to get the country mirror from the current locale "
  58.   # special cases go here
  59.   lang_mirror = { 'c'     : '',
  60.                 }
  61.   # no lang, no mirror
  62.   if not os.environ.has_key('LANG'):
  63.     return ''
  64.   lang = os.environ['LANG'].lower()
  65.   # check if it is a special case
  66.   if lang_mirror.has_key(lang[:5]):
  67.     return lang_mirror[lang[:5]]
  68.   # now check for the most comon form (en_US.UTF-8)
  69.   if "_" in lang:
  70.     country = lang.split(".")[0].split("_")[1]
  71.     if "@" in country:
  72.        country = country.split("@")[0]
  73.     return country+"."
  74.   else:
  75.     return lang[:2]+"."
  76.   return ''
  77.  
  78. def get_dist():
  79.   " return the codename of the current runing distro "
  80.   from subprocess import Popen, PIPE
  81.   p = Popen(["lsb_release","-c","-s"],stdout=PIPE)
  82.   res = p.wait()
  83.   if res != 0:
  84.     sys.stderr.write("lsb_release returned exitcode: %i\n" % res)
  85.     return "unknown distribution"
  86.   dist = p.stdout.readline().strip()
  87.   return dist
  88.  
  89. def url_downloadable(uri, debug_func=None):
  90.   """
  91.   helper that checks if the given uri exists and is downloadable
  92.   (supports optional debug_func function handler to support 
  93.    e.g. logging)
  94.  
  95.   Supports http (via HEAD) and ftp (via size request)
  96.   """
  97.   if debug_func:
  98.     debug_func("url_downloadable: %s" % uri)
  99.   import urlparse
  100.   (scheme, netloc, path, querry, fragment) = urlparse.urlsplit(uri)
  101.   if scheme == "http":
  102.     import httplib
  103.     try:
  104.       c = httplib.HTTPConnection(netloc)
  105.       c.request("HEAD", path)
  106.       res = c.getresponse()
  107.       if debug_func:
  108.         debug_func("url_downloadable result '%s'" % res.status)
  109.       res.close()
  110.       if res.status == 200:
  111.         return True
  112.     except Exception, e:
  113.       debug_func("error from httplib: '%s'" % e)
  114.       return False
  115.   elif scheme == "ftp":
  116.     import ftplib
  117.     try:
  118.       f = ftplib.FTP(netloc)
  119.       f.login()
  120.       f.cwd(os.path.dirname(path))
  121.       size = f.size(os.path.basename(path))
  122.       f.quit()
  123.       if debug_func:
  124.         debug_func("ftplib.size() returned: %s" % size)
  125.       if size != 0:
  126.         return True
  127.     except Exception, e:
  128.       if debug_func:
  129.         debug_func("error from ftplib: '%s'" % e)
  130.       return False
  131.   return False
  132.  
  133. def init_proxy(gconfclient=None):
  134.   """ init proxy settings 
  135.  
  136.   * first check for http_proxy environment (always wins),
  137.   * then check the apt.conf http proxy, 
  138.   * then look into synaptics conffile
  139.   * then into gconf  (if gconfclient was supplied)
  140.   """
  141.   SYNAPTIC_CONF_FILE = "/root/.synaptic/synaptic.conf"
  142.   proxy = None
  143.   # generic apt config wins
  144.   apt_pkg.InitConfig()
  145.   if apt_pkg.Config.Find("Acquire::http::Proxy") != '':
  146.     proxy = apt_pkg.Config.Find("Acquire::http::Proxy")
  147.   # then synaptic
  148.   elif os.path.exists(SYNAPTIC_CONF_FILE):
  149.     cnf = apt_pkg.newConfiguration()
  150.     apt_pkg.ReadConfigFile(cnf, SYNAPTIC_CONF_FILE)
  151.     use_proxy = cnf.FindB("Synaptic::useProxy", False)
  152.     if use_proxy:
  153.       proxy_host = cnf.Find("Synaptic::httpProxy")
  154.       proxy_port = str(cnf.FindI("Synaptic::httpProxyPort"))
  155.       if proxy_host and proxy_port:
  156.         proxy = "http://%s:%s/" % (proxy_host, proxy_port)
  157.   # then gconf
  158.   elif gconfclient:
  159.     try: # see LP: #281248
  160.       if gconfclient.get_bool("/system/http_proxy/use_http_proxy"):
  161.         host = gconfclient.get_string("/system/http_proxy/host")
  162.         port = gconfclient.get_int("/system/http_proxy/port")
  163.         use_auth = gconfclient.get_bool("/system/http_proxy/use_authentication")
  164.         if host and port:
  165.           if use_auth:
  166.             auth_user = gconfclient.get_string("/system/http_proxy/authentication_user")
  167.             auth_pw = gconfclient.get_string("/system/http_proxy/authentication_password")
  168.             proxy = "http://%s:%s@%s:%s/" % (auth_user,auth_pw,host, port)
  169.           else:
  170.             proxy = "http://%s:%s/" % (host, port)
  171.     except Exception, e:
  172.       print "error from gconf: %s" % e
  173.   # if we have a proxy, set it
  174.   if proxy:
  175.     # basic verification
  176.     if not proxy.startswith("http://"):
  177.       return
  178.     proxy_support = urllib2.ProxyHandler({"http":proxy})
  179.     opener = urllib2.build_opener(proxy_support)
  180.     urllib2.install_opener(opener)
  181.     os.putenv("http_proxy",proxy)
  182.  
  183. def on_battery():
  184.   """
  185.   Check a dbus signal to org.gnome.PowerManager to not suspend
  186.   the system, this is to support upgrades from pre-gutsy g-p-m
  187.   """
  188.   try:
  189.     import dbus
  190.     bus = dbus.Bus(dbus.Bus.TYPE_SYSTEM)
  191.     devobj = bus.get_object('org.freedesktop.DeviceKit.Power', 
  192.                             '/org/freedesktop/DeviceKit/Power')
  193.     dev = dbus.Interface(devobj, "org.freedesktop.DBus.Properties")
  194.     return dev.Get("org.freedesktop.DeviceKit.Power", "on_battery")
  195.   except Exception, e:
  196.     #import sys
  197.     #print >>sys.stderr, "on_battery returned error: ", e
  198.     return False
  199.  
  200. def _inhibit_sleep_old_interface():
  201.   """
  202.   Send a dbus signal to org.gnome.PowerManager to not suspend
  203.   the system, this is to support upgrades from pre-gutsy g-p-m
  204.   """
  205.   import dbus
  206.   bus = dbus.Bus(dbus.Bus.TYPE_SESSION)
  207.   devobj = bus.get_object('org.gnome.PowerManager', 
  208.                           '/org/gnome/PowerManager')
  209.   dev = dbus.Interface(devobj, "org.gnome.PowerManager")
  210.   cookie = dev.Inhibit('UpdateManager', 'Updating system')
  211.   return (dev, cookie)
  212.  
  213. def _inhibit_sleep_new_interface():
  214.   """
  215.   Send a dbus signal to gnome-power-manager to not suspend
  216.   the system
  217.   """
  218.   import dbus
  219.   bus = dbus.Bus(dbus.Bus.TYPE_SESSION)
  220.   devobj = bus.get_object('org.freedesktop.PowerManagement', 
  221.                           '/org/freedesktop/PowerManagement/Inhibit')
  222.   dev = dbus.Interface(devobj, "org.freedesktop.PowerManagement.Inhibit")
  223.   cookie = dev.Inhibit('UpdateManager', 'Updating system')
  224.   return (dev, cookie)
  225.  
  226. def inhibit_sleep():
  227.   """
  228.   Send a dbus signal to power-manager to not suspend
  229.   the system, try both the new freedesktop and the
  230.   old gnome dbus interface
  231.   """
  232.   try:
  233.     return _inhibit_sleep_old_interface()
  234.   except Exception, e:
  235.     try:
  236.       return _inhibit_sleep_new_interface()
  237.     except Exception, e:
  238.       #print "could not send the dbus Inhibit signal: %s" % e
  239.       return (False, False)
  240.  
  241. def allow_sleep(dev, cookie):
  242.   """Send a dbus signal to gnome-power-manager to allow a suspending
  243.   the system"""
  244.   try:
  245.     dev.UnInhibit(cookie)
  246.   except Exception, e:
  247.     print "could not send the dbus UnInhibit signal: %s" % e
  248.  
  249.  
  250. def str_to_bool(str):
  251.   if str == "0" or str.upper() == "FALSE":
  252.     return False
  253.   return True
  254.  
  255. def utf8(str):
  256.   return unicode(str, 'latin1').encode('utf-8')
  257.  
  258. def error(parent, summary, message):
  259.   import gtk
  260.   d = gtk.MessageDialog(parent=parent,
  261.                         flags=gtk.DIALOG_MODAL,
  262.                         type=gtk.MESSAGE_ERROR,
  263.                         buttons=gtk.BUTTONS_CLOSE)
  264.   d.set_markup("<big><b>%s</b></big>\n\n%s" % (summary, message))
  265.   d.realize()
  266.   d.window.set_functions(gtk.gdk.FUNC_MOVE)
  267.   d.set_title("")
  268.   res = d.run()
  269.   d.destroy()
  270.   return False
  271.  
  272. def humanize_size(bytes):
  273.     """
  274.     Convert a given size in bytes to a nicer better readable unit
  275.     """
  276.     if bytes == 0:
  277.         # TRANSLATORS: download size is 0
  278.         return _("0 KB")
  279.     elif bytes < 1024:
  280.         # TRANSLATORS: download size of very small updates
  281.         return _("1 KB")
  282.     elif bytes < 1024 * 1024:
  283.         # TRANSLATORS: download size of small updates, e.g. "250 KB"
  284.         return locale.format(_("%.0f KB"), bytes/1024)
  285.     else:
  286.         # TRANSLATORS: download size of updates, e.g. "2.3 MB"
  287.         return locale.format(_("%.1f MB"), bytes / 1024 / 1024)
  288.  
  289. if __name__ == "__main__":
  290.   #print mirror_from_sources_list()
  291.   print on_battery()
  292.